import WebSocket from 'ws';

// 创建WebSocket连接
const ws = new WebSocket('ws://localhost:3000');

// 连接建立时的处理
ws.on('open', () => {
    console.log('已连接到服务器');

    // 发送初始化命令
    ws.send(JSON.stringify({
        command: 'init',
        params: {
            center: [39.9042, 116.4074], // 北京坐标
            zoom: 1
        }
    }));

    // 发送绘制星形命令
    ws.send(JSON.stringify({
        command: 'drawStar',
        params: {
            center: [39.9042, 116.4074],
            radius: 1
        }
    }));

    // 发送缩放命令
    ws.send(JSON.stringify({
        command: 'zoomToStar',
        params: {
            center: [39.9042, 116.4074],
            radius: 1
        }
    }));

    // 等待一小段时间后关闭连接
    setTimeout(() => {
        console.log('所有命令已发送，正在关闭连接...');
        ws.close();
    }, 1000);
});

// 接收消息的处理
ws.on('message', (data) => {
    const response = JSON.parse(data);
    console.log('收到服务器响应:', response);
});

// 错误处理
ws.on('error', (error) => {
    console.error('WebSocket错误:', error);
    process.exit(1);
});

// 连接关闭的处理
ws.on('close', () => {
    console.log('连接已关闭');
    process.exit(0);
}); 